//给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。 
//
// 子数组是数组中元素的连续非空序列。 
//
// 
//
// 示例 1： 
//
// 
//输入：nums = [1,1,1], k = 2
//输出：2
// 
//
// 示例 2： 
//
// 
//输入：nums = [1,2,3], k = 3
//输出：2
// 
//
// 
//
// 提示： 
//
// 
// 1 <= nums.length <= 2 * 10⁴ 
// -1000 <= nums[i] <= 1000 
// -10⁷ <= k <= 10⁷ 
// 
//
// Related Topics 数组 哈希表 前缀和 👍 2173 👎 0

package leetcode.editor.cn;

import java.util.Arrays;
import java.util.HashMap;

//java:和为 K 的子数组
public class Q0560SubarraySumEqualsK {
    public static void main(String[] args) {
        Solution solution = new Q0560SubarraySumEqualsK().new Solution();
        solution.subarraySum(new int[]{3, 4, 7, 2, -3, 1, 4, 2, -6}, 7);
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        public int subarraySum(int[] nums, int k) {
            if (nums == null || nums.length == 0 || nums.length > 20000) return 0;
            int n = nums.length, res = 0;
            int sum = 0;
            HashMap<Integer, Integer> mp = new HashMap<>();
            mp.put(0, 1);
            for (int i = 0; i < n; i++) {
                sum += nums[i];
                if (mp.containsKey(sum - k)) {
                    res += mp.get(sum - k);
                }
                mp.put(sum, mp.getOrDefault(sum, 0) + 1);
            }
            return res;
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}